Home > biomechZoo > Toolbox > Machine Learning > model_train.m

model_train

PURPOSE ^

Mdl=model_train(ml_data, Model_name) trains a classification model

SYNOPSIS ^

function Mdl=model_train(ml_data, Model_name)

DESCRIPTION ^

 Mdl=model_train(ml_data, Model_name) trains a classification model

 ARGUMENTS
   ml_data        ...   struct, struct returned with train_test_split
   Model_name     ...   string,'BDT'     --> Binary Classification Tree
                          'NBayes'  --> Naive bayes
                          'knn'     --> k-Nearest Neighbor Classifier
                          'Bsvm'    --> Binary support vector machines
                          'Blinear' --> Binary Linear Classification
                          'Bkernel' --> Binary Kernel Classification
                          'Msvm'    --> Multiclass support vector machines
                          'FF'      --> Forward Feed Neural network
                          'LSTM'    --> Long short-term memory
                          'BiLS'  --> Bidirectional Long short-term memory
                          'CNN'     --> Convolutional neural network
 RETURNS
   Mdl         ...   Trained model

CROSS-REFERENCE INFORMATION ^

This function calls: This function is called by:
Generated on Sun 10-Mar-2024 22:39:06 by m2html © 2005